import numpy as np


def eye(size: int):
    return np.eye(size)


if __name__ == '__main__':
    from pprint import pprint

    pprint(eye(5))
